#include "GobangAI.h"
#include "Predictor.h"

CGobangAI::CGobangAI():difficulty(1){
}

CPoint CGobangAI::Calc(CGobangDoc* doc){
	if(doc->IsBlackChess()){
		Predictor predictor(doc,BLACK);
	}
	else{
		Predictor predictor(doc,WHITE);
	}

	return predictor.getPrediction(difficulty);
}

bool CGobangAI::SetDifficulty(int difficulty){
	if(difficulty>0){
		this->difficulty=difficulty;
		return true;
	}
	else{
		return false;
	}
}
